cd ../../

gpus=$1
ids_bsz=24

echo "Run baseline."
for od_size in 24 48 96
do
    python sed_main.py   \
        --gpus "${gpus}," \
        --exp_name "sed/baseline_odMT/o${od_size}_i0" \
        --batch_sizes "[0, 12, 12, 0, ${od_size}]" 
done
